import pandas as pd

# --- 1. 定义文件路径 ---
input_filename = 'price_data.csv'
output_filename = 'wuhan_price.csv'

# --- 2. 读取 CSV 文件 ---
try:
    # 使用 pandas 读取 CSV 文件到 DataFrame 中
    # engine='python' 有时可以帮助解决编码问题，如果文件中有中文
    df = pd.read_csv(input_filename, engine='python', encoding='utf-8-sig')
    print(f"成功读取文件: {input_filename}")
    print(f"原始数据共有 {len(df)} 行。")

except FileNotFoundError:
    print(f"错误: 文件 '{input_filename}' 未找到。请检查文件名和路径是否正确。")
    exit()

# --- 3. 筛选数据 ---
# 保留 '城市' 列中值为 '武汉' 的所有行
# .copy() 可以避免后续操作中可能出现的 SettingWithCopyWarning
wuhan_df = df[df['城市'] == '武汉'].copy()

# --- 4. 保存为新文件 ---
if not wuhan_df.empty:
    # 将筛选后的 DataFrame 保存到新的 CSV 文件
    # index=False 表示不将 DataFrame 的索引写入到 CSV 文件中
    wuhan_df.to_csv(output_filename, index=False, encoding='utf-8-sig')
    print(f"筛选完成！包含 '武汉' 的数据共有 {len(wuhan_df)} 行。")
    print(f"数据已成功保存到新文件: {output_filename}")
else:
    print("在原始数据中没有找到任何 '城市' 为 '武汉' 的记录。")